/*! ******************************************************************************
 *
 * Created on 2021年11月29日
 *
 * Copyright(c) YuZhou Big Data Laboratory CO.,Ltd, 2022.  All rights reserved.
 *
 *******************************************************************************
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 ******************************************************************************/
package org.pentaho.di.ui.spoon.navigation.view.subtree;

import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.MessageBox;
import org.eclipse.swt.widgets.TreeItem;
import org.pentaho.di.base.AbstractMeta;
import org.pentaho.di.core.database.DatabaseMeta;
import org.pentaho.di.core.exception.KettleException;
import org.pentaho.di.i18n.BaseMessages;
import org.pentaho.di.repository.KettleRepositoryLostException;
import org.pentaho.di.ui.core.dialog.ErrorDialog;
import org.pentaho.di.ui.core.gui.GUIResource;
import org.pentaho.di.ui.spoon.DatabasesCollector;
import org.pentaho.di.ui.spoon.Spoon;
import org.pentaho.di.ui.spoon.SpoonLifecycleListener.SpoonLifeCycleEvent;
import org.pentaho.di.ui.spoon.SpoonPluginManager;
import org.pentaho.di.ui.spoon.navigation.view.Subtree;
import org.pentaho.di.ui.spoon.navigation.view.MetaTree;

/**
 * DB连接子树
 * 
 * @author hs@yzbdl.ac.cn
 *
 */
public class DbConnectionsSubtree extends Subtree {
	private static Class<?> PKG = Spoon.class;

	public DbConnectionsSubtree(MetaTree parent, AbstractMeta meta) {
		super(parent);

		DatabasesCollector collector = new DatabasesCollector(meta, this.getSpoonRef().getRep());
		try {
			try {
				collector.collectDatabases();
			} catch (KettleException e) {
				if (e.getCause() instanceof KettleRepositoryLostException) {
					handleRepositoryLost((KettleRepositoryLostException) e.getCause());
					collector = new DatabasesCollector(meta, null);
					collector.collectDatabases();
				} else {
					throw e;
				}
			}
		} catch (KettleException e) {
			new ErrorDialog(this.getSpoonRef().getShell(), BaseMessages.getString(PKG, "Spoon.ErrorDialog.Title"),
					BaseMessages.getString(PKG, "Spoon.ErrorDialog.ErrorFetchingFromRepo.DbConnections"), e);
		}

		for (String dbName : collector.getDatabaseNames()) {
			if (!this.getSpoonRef().filterMatch(dbName)) {
				continue;
			}
			DatabaseMeta databaseMeta = collector.getMetaFor(dbName);

			TreeItem tiDb = new TreeItem(this.getView(), SWT.NONE);
			tiDb.setText(databaseMeta.getDisplayName());
			tiDb.setImage(GUIResource.getInstance().getImageConnectionTree());

			if (databaseMeta.isShared()) {
				tiDb.setFont(GUIResource.getInstance().getFontBold());
			}
		}
	}

	public void handleRepositoryLost(KettleRepositoryLostException e) {
		this.getSpoonRef().setRepository(null);
		warnRepositoryLost(e);
		SpoonPluginManager.getInstance().notifyLifecycleListeners(SpoonLifeCycleEvent.REPOSITORY_DISCONNECTED);
		this.getSpoonRef().setShellText();
		this.getSpoonRef().enableMenus();
	}

	private void warnRepositoryLost(KettleRepositoryLostException e) {
		MessageBox box = new MessageBox(this.getSpoonRef().getShell(), SWT.OK | SWT.ICON_WARNING);
		box.setText(BaseMessages.getString(PKG, "System.Warning"));
		box.setMessage(e.getPrefaceMessage());
		box.open();
	}

	/**
	 * (non-Javadoc)
	 * 
	 * @see org.pentaho.di.ui.spoon.navigation.view.Subtree#getName()
	 */
	@Override
	public String getName() {
		return BaseMessages.getString(PKG, "Spoon.STRING_CONNECTIONS");
	}

}
